export default {
  install (Vue) {
    Vue.directive('focus', {
      inserted: function (el) {
        // el = el.querySelector('input')
        // el.focus()
        if (el.nodeName === 'INPUT' || el.nodeName === 'TEXTAREA') {
          el.focus()
        } else {
          const inp = el.querySelector('input')
          const textarea = el.querySelector('textarea')
          if (inp || textarea) {
            inp && inp.focus()
            textarea && textarea.focus()
          } else {
            console.error('请在输入框或文本域内使用v-focus方法')
          }
        }
      }
    })
  }
}
